import paddle
from models.EncoderCNN import EncoderCNN

class Net(paddle.nn.Layer):
    def __init__(self):
        super().__init__()
        resnet = paddle.vision.models.resnet152(pretrained=True)
        #  self.net = paddle.nn.Sequential(*(list(resnet.children())[:-1]))
        self.net = resnet

    def forward(self, image):
        return self.net(image)

blank_batch = paddle.zeros((64, 3, 224, 224), dtype='float32')

#  cnn = EncoderCNN()
net = Net()

print(net(blank_batch))
